#include "uwsgi.h"

extern struct uwsgi_server uwsgi;

static size_t get_content_length(char *buf, uint16_t size) {
	int i;
	size_t val = 0;
	for(i=0;i<size;i++) {
		if (buf[i] >= '0' && buf[i] <= '9') {
			val = (val*10) + (buf[i] - '0');
			continue;
		}
		break;
	}

	return val;
}


int set_http_date(time_t t, char *header, int header_len, char *dst, int last) {

	static char  *week[] = { "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat" };
	static char  *months[] = {
				"Jan", "Feb", "Mar", "Apr",
				"May", "Jun", "Jul", "Aug",
				"Sep", "Oct", "Nov", "Dec"
			};

	struct tm *hdtm = gmtime(&t);

	if (last) {
		return snprintf(dst, 36+header_len, "%.*s: %s, %02d %s %4d %02d:%02d:%02d GMT\r\n\r\n",
			header_len, header,
			week[hdtm->tm_wday], hdtm->tm_mday,
			months[hdtm->tm_mon], hdtm->tm_year+1900,
			hdtm->tm_hour, hdtm->tm_min, hdtm->tm_sec);
	}

	return snprintf(dst, 34+header_len, "%.*s: %s, %02d %s %4d %02d:%02d:%02d GMT\r\n",
		header_len, header,
		week[hdtm->tm_wday], hdtm->tm_mday,
		months[hdtm->tm_mon], hdtm->tm_year+1900,
		hdtm->tm_hour, hdtm->tm_min, hdtm->tm_sec);
}

void uwsgi_add_expires_type(struct wsgi_request *wsgi_req, char *mime_type, int mime_type_len, struct stat *st) {

	struct uwsgi_dyn_dict *udd = uwsgi.static_expires_type;
	time_t now = wsgi_req->start_of_request.tv_sec;
	// Expires+34+1
	char expires[42];

	while(udd) {
		if (!uwsgi_strncmp(udd->key, udd->keylen, mime_type, mime_type_len)) {
			int delta = uwsgi_str_num(udd->value, udd->vallen);
			int size = set_http_date(now+delta, "Expires", 7, expires, 0);
			if (size > 0) {
				wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, expires, size);
				wsgi_req->header_cnt++;
			}
			return;
		}
		udd = udd->next;
	}

	udd = uwsgi.static_expires_type_mtime;
	while(udd) {
		if (!uwsgi_strncmp(udd->key, udd->keylen, mime_type, mime_type_len)) {
			int delta = uwsgi_str_num(udd->value, udd->vallen);
			int size = set_http_date(st->st_mtime+delta, "Expires", 7, expires, 0);
			if (size > 0) {
				wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, expires, size);
				wsgi_req->header_cnt++;
			}
			return;
		}
		udd = udd->next;
	}
}

#ifdef UWSGI_PCRE
void uwsgi_add_expires(struct wsgi_request *wsgi_req, char *filename, int filename_len, struct stat *st) {

        struct uwsgi_dyn_dict *udd = uwsgi.static_expires;
        time_t now = wsgi_req->start_of_request.tv_sec;
        // Expires+34+1
        char expires[42];

        while(udd) {
                if (uwsgi_regexp_match(udd->pattern, udd->pattern_extra, filename, filename_len) >= 0) {
                        int delta = uwsgi_str_num(udd->value, udd->vallen);
                        int size = set_http_date(now+delta, "Expires", 7, expires, 0);
                        if (size > 0) {
                                wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, expires, size);
                                wsgi_req->header_cnt++;
                        }
                        return;
                }
                udd = udd->next;
        }

        udd = uwsgi.static_expires_mtime;
        while(udd) {
		if (uwsgi_regexp_match(udd->pattern, udd->pattern_extra, filename, filename_len) >= 0) {
                        int delta = uwsgi_str_num(udd->value, udd->vallen);
                        int size = set_http_date(st->st_mtime+delta, "Expires", 7, expires, 0);
                        if (size > 0) {
                                wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, expires, size);
                                wsgi_req->header_cnt++;
                        }
                        return;
                }
                udd = udd->next;
        }
}

#endif


// only RFC 1123 is supported
time_t parse_http_date(char *date, uint16_t len) {

	struct tm hdtm;

	if (len != 29 && date[3] != ',') return 0;

	hdtm.tm_mday = uwsgi_str2_num(date+5);

	switch(date[8]) {
		case 'J':
			if (date[9] == 'a') {
				hdtm.tm_mon = 0;
				break;
			}

			if (date[9] == 'u') {
				if (date[10] == 'n') {
					hdtm.tm_mon = 5;
					break;
				}

				if (date[10] == 'l') {
					hdtm.tm_mon = 6;
					break;
				}

				return 0;
			}

			return 0;

		case 'F':
			hdtm.tm_mon = 1;
			break;

		case 'M':
			if (date[9] != 'a') return 0;

			if (date[10] == 'r') {
				hdtm.tm_mon = 2;
				break;
			}

			if (date[10] == 'y') {
				hdtm.tm_mon = 4;
				break;
			}

			return 0;

		case 'A':
			if (date[10] == 'r') {
				hdtm.tm_mon = 3;
				break;
			}
			if (date[10] == 'g') {
				hdtm.tm_mon = 7;
				break;
			}
			return 0;

		case 'S':
			hdtm.tm_mon = 8;
			break;

		case 'O':
			hdtm.tm_mon = 9;
			break;

		case 'N':
			hdtm.tm_mon = 10;

		case 'D':
			hdtm.tm_mon = 11;
			break;
		default:
			return 0;
	}

	hdtm.tm_year = uwsgi_str4_num(date+12)-1900;

	hdtm.tm_hour = uwsgi_str2_num(date+17);
	hdtm.tm_min = uwsgi_str2_num(date+20);
	hdtm.tm_sec = uwsgi_str2_num(date+23);

	return timegm(&hdtm);
	
}

ssize_t send_udp_message(uint8_t modifier1, uint8_t modifier2, char *host, char *message, uint16_t message_size) {

	int fd;
	struct sockaddr_in udp_addr;
	char *udp_port;
	ssize_t ret;

	struct uwsgi_header *uh;

	if (message) {
		uh = (struct uwsgi_header *) message;
	}
	else {
		uh = (struct uwsgi_header *) uwsgi_malloc(4);
	}

	udp_port = strchr(host, ':');
	if (udp_port == NULL) {
		return -1;
	}

	udp_port[0] = 0; 

	fd = socket(AF_INET, SOCK_DGRAM, 0);
	if (fd < 0) {
		uwsgi_error("socket()");
		return -1;
	}

	memset(&udp_addr, 0, sizeof(struct sockaddr_in));
	udp_addr.sin_family = AF_INET;
	udp_addr.sin_port = htons(atoi(udp_port+1));
	udp_addr.sin_addr.s_addr = inet_addr(host);

	uh->modifier1 = modifier1;
#ifdef __BIG_ENDIAN__
	uh->pktsize = uwsgi_swap16(message_size);
#else
	uh->pktsize = message_size;
#endif
	uh->modifier2 = modifier2;

	ret = sendto(fd, (char *) uh, message_size+4, 0, (struct sockaddr *) &udp_addr, sizeof(udp_addr));
	if (ret < 0) {
		uwsgi_error("sendto()");
	}
	close(fd);

	udp_port[0] = ':';

	if ((char *)uh != message) {
		free(uh);
	}

	return ret;
	
}

int uwsgi_enqueue_message(char *host, int port, uint8_t modifier1, uint8_t modifier2, char *message, int size, int timeout) {

	struct pollfd uwsgi_poll;
	struct sockaddr_in uws_addr;
	int cnt;
	struct uwsgi_header uh;

	if (!timeout)
		timeout = 1;

	if (size > 0xffff) {
		uwsgi_log( "invalid object (marshalled) size\n");
		return -1;
	}

	uwsgi_poll.fd = socket(AF_INET, SOCK_STREAM, 0);
	if (uwsgi_poll.fd < 0) {
		uwsgi_error("socket()");
		return -1;
	}

	memset(&uws_addr, 0, sizeof(struct sockaddr_in));
	uws_addr.sin_family = AF_INET;
	uws_addr.sin_port = htons(port);
	uws_addr.sin_addr.s_addr = inet_addr(host);

	uwsgi_poll.events = POLLIN;

	if (timed_connect(&uwsgi_poll, (const struct sockaddr *) &uws_addr, sizeof(struct sockaddr_in), timeout, 0)) {
		uwsgi_error("connect()");
		close(uwsgi_poll.fd);
		return -1;
	}

	uh.modifier1 = modifier1;
	uh.pktsize = (uint16_t) size;
	uh.modifier2 = modifier2;

	cnt = write(uwsgi_poll.fd, &uh, 4);
	if (cnt != 4) {
		uwsgi_error("write()");
		close(uwsgi_poll.fd);
		return -1;
	}

	cnt = write(uwsgi_poll.fd, message, size);
	if (cnt != size) {
		uwsgi_error("write()");
		close(uwsgi_poll.fd);
		return -1;
	}

	return uwsgi_poll.fd;
}



ssize_t uwsgi_send_message(int fd, uint8_t modifier1, uint8_t modifier2, char *message, uint16_t size, int pfd, ssize_t plen, int timeout) {

	struct pollfd uwsgi_mpoll;
	ssize_t cnt;
	struct uwsgi_header uh;
	char buffer[4096];
	ssize_t ret = 0;
	int pret;
	struct msghdr msg;
	struct iovec  iov [1];
	union {
		struct cmsghdr cmsg;
		char control [CMSG_SPACE (sizeof (int))];
	} msg_control;
	struct cmsghdr *cmsg;

	if (!timeout) timeout = uwsgi.shared->options[UWSGI_OPTION_SOCKET_TIMEOUT];

	uh.modifier1 = modifier1;
	uh.pktsize = size;
	uh.modifier2 = modifier2;

	if (pfd >= 0 && plen == -1) {
		// pass the fd
		iov[0].iov_base = &uh;
		iov[0].iov_len = 4;
		
		msg.msg_name    = NULL;
		msg.msg_namelen = 0;
		msg.msg_iov     = iov;
		msg.msg_iovlen  = 1;
		msg.msg_flags   = 0;

		msg.msg_control    = &msg_control;
		msg.msg_controllen = sizeof (msg_control);

		cmsg = CMSG_FIRSTHDR (&msg);
		cmsg->cmsg_len   = CMSG_LEN (sizeof (int));
		cmsg->cmsg_level = SOL_SOCKET;
		cmsg->cmsg_type  = SCM_RIGHTS;

		memcpy(CMSG_DATA(cmsg), &pfd, sizeof(int));
		
#ifdef UWSGI_DEBUG
		uwsgi_log("passing fd\n");
#endif
		cnt = sendmsg(fd, &msg, 0);	
	}
	else {
		cnt = write(fd, &uh, 4);
	}
	if (cnt != 4) {
		uwsgi_error("write()");
		return -1;
	}

	ret += cnt;

	cnt = write(fd, message, size);
	if (cnt != size) {
		uwsgi_error("write()");
		return -1;
	}

	ret += cnt;

	// transfer data from one socket to another
	if (pfd >= 0 && plen > 0) {
		uwsgi_mpoll.fd = pfd;
		uwsgi_mpoll.events = POLLIN;
		
		while(plen > 0) {
			pret = poll(&uwsgi_mpoll, 1, timeout*1000);
			if (pret < 0) {
				uwsgi_error("poll()");
				return -1;
			}
			else if (pret == 0) {
				uwsgi_log("timeout waiting for socket data\n");
				return -1;
			}
			else {
				cnt = read(pfd, buffer, UMIN(4096, plen));
				if (cnt < 0) {
					uwsgi_error("read()");
					return -1;
				}
				else if (cnt == 0) {
					return ret;
				}	
				// send to peer
				if (write(fd, buffer, cnt) != cnt) {
					uwsgi_error("write()");
					return -1;
				}
				ret += cnt;
				plen -= cnt;	
			}
		}
	}


	return ret;
}

int uwsgi_read_response(int fd, struct uwsgi_header *uh, int timeout, char **buf) {

	char *ptr = (char *) uh;
	size_t remains = 4;
	int ret = -1;
	int rlen;
	ssize_t len;

	while(remains > 0) {
                rlen = uwsgi_waitfd(fd, timeout);
                if (rlen > 0) {
                        len = read(fd, ptr, remains);
                        if (len <= 0) break;
                        remains -= len;
                        ptr += len;
                        if (remains == 0) {
                                ret = uh->modifier2;
				break;
                        }
                        continue;
                }
		// timed out ?
		else if (ret == 0) ret = -2;
                break;
        }

	if (buf && uh->pktsize > 0) {
		if (*buf == NULL)
			*buf = uwsgi_malloc(uh->pktsize);
		remains = uh->pktsize;
		ptr = *buf;
		ret = -1;
		while(remains > 0) {
			rlen = uwsgi_waitfd(fd, timeout);
			if (rlen > 0) {
				len = read(fd, ptr, remains);
				if (len <= 0) break;
				remains -= len;
				ptr += len;
				if (remains == 0) {
					ret = uh->modifier2;
					break;
				}
				continue;
			}
			// timed out ?
			else if (ret == 0) ret = -2;
			break;
		}
	}

	return ret;
}

int uwsgi_parse_packet(struct wsgi_request *wsgi_req, int timeout) {
	int rlen;
	int status = UWSGI_AGAIN;

	if (!timeout)
		timeout = 1;

	while(status == UWSGI_AGAIN) {
		rlen = poll(&wsgi_req->poll, 1, timeout * 1000);
		if (rlen < 0) {
			uwsgi_error("poll()");
			exit(1);
		}
		else if (rlen == 0) {
			uwsgi_log( "timeout. skip request.\n");
			//close(upoll->fd);
			return 0;
		}
		if (wsgi_req->socket) {
			status = wsgi_req->socket->proto(wsgi_req);
		}
		else {
			status = uwsgi_proto_uwsgi_parser(wsgi_req);
		}
		if (status < 0) {
			if (status == -1)
				uwsgi_log_verbose("error parsing request\n");
			else if (status == -2)
				uwsgi_log_verbose("open-close packet (ping/check) received\n");
			//close(upoll->fd);
			return 0;
		}
	}

	return 1;
}

int uwsgi_parse_array(char *buffer, uint16_t size, char **argv, uint16_t argvs[], uint8_t *argc) {

	char *ptrbuf, *bufferend;
	uint16_t strsize = 0;
	
	uint8_t max = *argc;
	*argc = 0;

        ptrbuf = buffer;
        bufferend = ptrbuf + size;

	while (ptrbuf < bufferend && *argc < max) {
                if (ptrbuf + 2 < bufferend) {
                        memcpy(&strsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
                        strsize = uwsgi_swap16(strsize);
#endif

                        ptrbuf += 2;
                        /* item cannot be null */
                        if (!strsize) continue;

                        if (ptrbuf + strsize <= bufferend) {
                                // item
				argv[*argc] = uwsgi_cheap_string(ptrbuf, strsize);
				argvs[*argc] = strsize;
#ifdef UWSGI_DEBUG
				uwsgi_log("arg %s\n", argv[*argc]);
#endif
                                ptrbuf += strsize;
				*argc = *argc + 1;
			}
			else {
				uwsgi_log( "invalid uwsgi array. skip this var.\n");
                        	return -1;
			}
		}
		else {
			uwsgi_log( "invalid uwsgi array. skip this request.\n");
                        return -1;
		}
	}
	

	return 0;
}

int uwsgi_simple_parse_vars(struct wsgi_request *wsgi_req, char *ptrbuf, char *bufferend) {
	
	uint16_t strsize;

	while (ptrbuf < bufferend) {
                if (ptrbuf + 2 < bufferend) {
                        memcpy(&strsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
                        strsize = uwsgi_swap16(strsize);
#endif
                        /* key cannot be null */
                        if (!strsize) {
                                uwsgi_log( "uwsgi key cannot be null. skip this request.\n");
                                return -1;
                        }

                        ptrbuf += 2;
                        if (ptrbuf + strsize < bufferend) {
                                // var key
                                wsgi_req->hvec[wsgi_req->var_cnt].iov_base = ptrbuf;
                                wsgi_req->hvec[wsgi_req->var_cnt].iov_len = strsize;
                                ptrbuf += strsize;
                                // value can be null (even at the end) so use <=
                                if (ptrbuf + 2 <= bufferend) {
                                        memcpy(&strsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
                                        strsize = uwsgi_swap16(strsize);
#endif
                                        ptrbuf += 2;
                                        if (ptrbuf + strsize <= bufferend) {
                                                
                                                if (wsgi_req->var_cnt < uwsgi.vec_size - (4 + 1)) {
                                                        wsgi_req->var_cnt++;
                                                }
                                                else {
                                                        uwsgi_log( "max vec size reached. skip this header.\n");
                                                        return -1;
                                                }
                                                // var value
                                                wsgi_req->hvec[wsgi_req->var_cnt].iov_base = ptrbuf;
                                                wsgi_req->hvec[wsgi_req->var_cnt].iov_len = strsize;

                                                if (wsgi_req->var_cnt < uwsgi.vec_size - (4 + 1)) {
                                                        wsgi_req->var_cnt++;
                                                }
                                                else {
                                                        uwsgi_log( "max vec size reached. skip this var.\n");
                                                        return -1;
                                                }
                                                ptrbuf += strsize;
                                        }
                                        else {
                                                uwsgi_log("invalid uwsgi request (current strsize: %d). skip.\n", strsize);
                                                return -1;
                                        }
                                }
                                else {
                                        uwsgi_log("invalid uwsgi request (current strsize: %d). skip.\n", strsize);
                                        return -1;
                                }
			}
		}
	}

	return 0;
}

int uwsgi_parse_vars(struct wsgi_request *wsgi_req) {

	char *buffer = wsgi_req->buffer;

	char *ptrbuf, *bufferend;

	uint16_t strsize = 0;
	struct uwsgi_dyn_dict *udd;

	ptrbuf = buffer;
	bufferend = ptrbuf + wsgi_req->uh.pktsize;
	int i, script_name= -1;

	/* set an HTTP 500 status as default */
	wsgi_req->status = 500;

	// skip if already parsed
	if (wsgi_req->parsed) return 0;

	// has the protocol already parsed the request ?
	if (wsgi_req->uri_len > 0) {
		wsgi_req->parsed = 1;
		i = uwsgi_simple_parse_vars(wsgi_req, ptrbuf, bufferend);
		if (i == 0) goto next;
		return i;
	}

	wsgi_req->parsed = 1;
	wsgi_req->path_info_pos = -1;

	while (ptrbuf < bufferend) {
		if (ptrbuf + 2 < bufferend) {
			memcpy(&strsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
			strsize = uwsgi_swap16(strsize);
#endif
			/* key cannot be null */
                        if (!strsize) {
                                uwsgi_log( "uwsgi key cannot be null. skip this var.\n");
                                return -1;
                        }
			
			ptrbuf += 2;
			if (ptrbuf + strsize < bufferend) {
				// var key
				wsgi_req->hvec[wsgi_req->var_cnt].iov_base = ptrbuf;
				wsgi_req->hvec[wsgi_req->var_cnt].iov_len = strsize;
				ptrbuf += strsize;
				// value can be null (even at the end) so use <=
				if (ptrbuf + 2 <= bufferend) {
					memcpy(&strsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
					strsize = uwsgi_swap16(strsize);
#endif
					ptrbuf += 2;
					if (ptrbuf + strsize <= bufferend) {
						//uwsgi_log("uwsgi %.*s = %.*s\n", wsgi_req->hvec[wsgi_req->var_cnt].iov_len, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, strsize, ptrbuf);
						if (!uwsgi_strncmp("SCRIPT_NAME", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->script_name = ptrbuf;
							wsgi_req->script_name_len = strsize;
							script_name = wsgi_req->var_cnt + 1;
#ifdef UWSGI_DEBUG
							uwsgi_debug("SCRIPT_NAME=%.*s\n", wsgi_req->script_name_len, wsgi_req->script_name);
#endif
						}
						else if (!uwsgi_strncmp("PATH_INFO", 9, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->path_info = ptrbuf;
							wsgi_req->path_info_len = strsize;
							wsgi_req->path_info_pos = wsgi_req->var_cnt + 1;
#ifdef UWSGI_DEBUG
							uwsgi_debug("PATH_INFO=%.*s\n", wsgi_req->path_info_len, wsgi_req->path_info);
#endif
						}
						else if (!uwsgi_strncmp("SERVER_PROTOCOL", 15, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->protocol = ptrbuf;
							wsgi_req->protocol_len = strsize;
						}
						else if (!uwsgi_strncmp("REQUEST_URI", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->uri = ptrbuf;
							wsgi_req->uri_len = strsize;
						}
						else if (!uwsgi_strncmp("QUERY_STRING", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->query_string = ptrbuf;
							wsgi_req->query_string_len = strsize;
						}
						else if (!uwsgi_strncmp("REQUEST_METHOD", 14, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->method = ptrbuf;
							wsgi_req->method_len = strsize;
						}
						else if (!uwsgi.log_x_forwarded_for && !uwsgi_strncmp("REMOTE_ADDR", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->remote_addr = ptrbuf;
							wsgi_req->remote_addr_len = strsize;
						}
						else if (!uwsgi_strncmp("REMOTE_USER", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->remote_user = ptrbuf;
							wsgi_req->remote_user_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_SCHEME", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->scheme = ptrbuf;
							wsgi_req->scheme_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_SCRIPT", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len )) {
							wsgi_req->script = ptrbuf;
							wsgi_req->script_len = strsize;
							wsgi_req->dynamic = 1;
						}
						else if (!uwsgi_strncmp("UWSGI_MODULE", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->module = ptrbuf;
							wsgi_req->module_len = strsize;
							wsgi_req->dynamic = 1;
						}
						else if (!uwsgi_strncmp("UWSGI_CALLABLE", 14, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->callable = ptrbuf;
							wsgi_req->callable_len = strsize;
							wsgi_req->dynamic = 1;
						}
						else if (!uwsgi_strncmp("UWSGI_PYHOME", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->pyhome = ptrbuf;
							wsgi_req->pyhome_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_CHDIR", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->chdir = ptrbuf;
							wsgi_req->chdir_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_APPID", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->appid = ptrbuf;
							wsgi_req->appid_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_FILE", 10, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->file = ptrbuf;
							wsgi_req->file_len = strsize;
							wsgi_req->dynamic = 1;
						}
						else if (!uwsgi_strncmp("UWSGI_POSTFILE", 14, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							char *postfile = uwsgi_concat2n(ptrbuf, strsize, "", 0);
							wsgi_req->async_post = fopen(postfile, "r");
							if (!wsgi_req->async_post) {
								uwsgi_error_open(postfile);
							}
							free(postfile);
							wsgi_req->body_as_file = 1;
						}
						else if (!uwsgi_strncmp("UWSGI_TOUCH_RELOAD", 18, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->touch_reload = ptrbuf;
							wsgi_req->touch_reload_len = strsize;
						}
						else if (uwsgi.cache_max_items > 0 && !uwsgi_strncmp("UWSGI_CACHE_GET", 15, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->cache_get = ptrbuf;
							wsgi_req->cache_get_len = strsize;
						}
						else if (!uwsgi_strncmp("UWSGI_SETENV", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							char *env_value = memchr(ptrbuf, '=', strsize);
							if (env_value) {
								env_value[0] = 0;
								env_value = uwsgi_concat2n(env_value+1, strsize-((env_value+1)-ptrbuf), "", 0);
								if (setenv(ptrbuf, env_value, 1)) {
									uwsgi_error("setenv()");
								}
								free(env_value);
							}
						}
						else if (!uwsgi_strncmp("SERVER_NAME", 11, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len) && !wsgi_req->host_len) {
							wsgi_req->host = ptrbuf;
							wsgi_req->host_len = strsize;
#ifdef UWSGI_DEBUG
							uwsgi_debug("SERVER_NAME=%.*s\n", wsgi_req->host_len, wsgi_req->host);
#endif
						}
						else if (!uwsgi_strncmp("HTTP_HOST", 9, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->host = ptrbuf;
							wsgi_req->host_len = strsize;
#ifdef UWSGI_DEBUG
							uwsgi_debug("HTTP_HOST=%.*s\n", wsgi_req->host_len, wsgi_req->host);
#endif
						}
						else if (!uwsgi_strncmp("HTTPS", 5, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->https = ptrbuf;
							wsgi_req->https_len = strsize;
						}
						else if (!uwsgi_strncmp("HTTP_IF_MODIFIED_SINCE", 22, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->if_modified_since = ptrbuf;
							wsgi_req->if_modified_since_len = strsize;
						}
						else if (!uwsgi_strncmp("HTTP_AUTHORIZATION", 18, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->authorization = ptrbuf;
							wsgi_req->authorization_len = strsize;
						}
						else if (!uwsgi_strncmp("DOCUMENT_ROOT", 13, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->document_root = ptrbuf;
							wsgi_req->document_root_len = strsize;
						}
						else if (uwsgi.log_x_forwarded_for && !uwsgi_strncmp("HTTP_X_FORWARDED_FOR", 20, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->remote_addr = ptrbuf;
							wsgi_req->remote_addr_len = strsize;
						}
						else if (!uwsgi_strncmp("CONTENT_TYPE", 12, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->content_type = ptrbuf;
							wsgi_req->content_type_len = strsize;
						}
						else if (!uwsgi_strncmp("CONTENT_LENGTH", 14, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
							wsgi_req->post_cl = get_content_length(ptrbuf, strsize);
							if (uwsgi.limit_post) {
                						if (wsgi_req->post_cl > uwsgi.limit_post) {
                        						uwsgi_log("Invalid (too big) CONTENT_LENGTH. skip.\n");
                        						return -1;
                						}
        						}

						}

						if (wsgi_req->var_cnt < uwsgi.vec_size - (4 + 1)) {
							wsgi_req->var_cnt++;
						}
						else {
							uwsgi_log( "max vec size reached. skip this var.\n");
							return -1;
						}
						// var value
						wsgi_req->hvec[wsgi_req->var_cnt].iov_base = ptrbuf;
						wsgi_req->hvec[wsgi_req->var_cnt].iov_len = strsize;
						//uwsgi_log("%.*s = %.*s\n", wsgi_req->hvec[wsgi_req->var_cnt-1].iov_len, wsgi_req->hvec[wsgi_req->var_cnt-1].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len, wsgi_req->hvec[wsgi_req->var_cnt].iov_base);
						if (wsgi_req->var_cnt < uwsgi.vec_size - (4 + 1)) {
							wsgi_req->var_cnt++;
						}
						else {
							uwsgi_log( "max vec size reached. skip this var.\n");
							return -1;
						}
						ptrbuf += strsize;
					}
					else {
						uwsgi_log("invalid uwsgi request (current strsize: %d). skip.\n", strsize);
						return -1;
					}
				}
				else {
					uwsgi_log("invalid uwsgi request (current strsize: %d). skip.\n", strsize);
					return -1;
				}
			}
		}
		else {
			uwsgi_log("invalid uwsgi request (current strsize: %d). skip.\n", strsize);
			return -1;
		}
	}

next:

	if (uwsgi.post_buffering > 0 && !wsgi_req->body_as_file && !wsgi_req->async_post) {
        	// read to disk if post_cl > post_buffering (it will eventually do upload progress...)
                if (wsgi_req->post_cl >= (size_t) uwsgi.post_buffering) {
                	if (!uwsgi_read_whole_body(wsgi_req, wsgi_req->post_buffering_buf, uwsgi.post_buffering_bufsize)) {
				wsgi_req->status = -1;
				return -1;	
                        }
			wsgi_req->body_as_file = 1;
		}
                // on tiny post use memory
                else {
                	if (!uwsgi_read_whole_body_in_mem(wsgi_req, wsgi_req->post_buffering_buf)) {
				wsgi_req->status = -1;
				return -1;	
                        }
		}
	}


	// check if data are available in the local cache
	if (wsgi_req->cache_get_len > 0) {
		uint64_t cache_value_size;
		char *cache_value = uwsgi_cache_get(wsgi_req->cache_get, wsgi_req->cache_get_len, &cache_value_size);
		if (cache_value && cache_value_size > 0) {
			wsgi_req->response_size = wsgi_req->socket->proto_write(wsgi_req, cache_value, cache_value_size);
			wsgi_req->status = -1;
			return -1;
		}
	}

	if (uwsgi.check_cache && wsgi_req->uri_len && wsgi_req->method_len == 3 &&
		wsgi_req->method[0] == 'G' && wsgi_req->method[1] == 'E' && wsgi_req->method[2] == 'T') {

		uint64_t cache_value_size;
		char *cache_value = uwsgi_cache_get(wsgi_req->uri, wsgi_req->uri_len, &cache_value_size);
		if (cache_value && cache_value_size > 0) {
			wsgi_req->response_size = wsgi_req->socket->proto_write(wsgi_req, cache_value, cache_value_size);
			wsgi_req->status = -1;
			return -1;
		}
	}

	if (uwsgi.manage_script_name) {
		if (uwsgi_apps_cnt > 0 && wsgi_req->path_info_len > 1 && wsgi_req->path_info_pos != -1) {
			// starts with 1 as the 0 app is the default (/) one
			int best_found = 0;
			char *orig_path_info = wsgi_req->path_info;
			int orig_path_info_len = wsgi_req->path_info_len;
			// if SCRIPT_NAME is not allocated, add a slot for it
			if (script_name == -1) {
				if (wsgi_req->var_cnt >= uwsgi.vec_size - (4 + 2)) {
					uwsgi_log( "max vec size reached. skip this var.\n");
                                        return -1;
				}
				wsgi_req->hvec[wsgi_req->var_cnt].iov_base = "SCRIPT_NAME";
                                wsgi_req->hvec[wsgi_req->var_cnt].iov_len = 11;
				wsgi_req->var_cnt++;
				script_name = wsgi_req->var_cnt;
				wsgi_req->hvec[script_name].iov_base = "";
				wsgi_req->hvec[script_name].iov_len = 0;
				wsgi_req->var_cnt++;
			}

			for(i=0;i<uwsgi_apps_cnt;i++) {
				//uwsgi_log("app mountpoint = %.*s\n", uwsgi_apps[i].mountpoint_len, uwsgi_apps[i].mountpoint);
				if (orig_path_info_len >= uwsgi_apps[i].mountpoint_len) {
					if (!uwsgi_startswith(orig_path_info, uwsgi_apps[i].mountpoint, uwsgi_apps[i].mountpoint_len) && uwsgi_apps[i].mountpoint_len > best_found) {
						best_found = uwsgi_apps[i].mountpoint_len;
						wsgi_req->script_name = uwsgi_apps[i].mountpoint;
						wsgi_req->script_name_len = uwsgi_apps[i].mountpoint_len;
						wsgi_req->path_info = orig_path_info+wsgi_req->script_name_len;
						wsgi_req->path_info_len = orig_path_info_len-wsgi_req->script_name_len;

						wsgi_req->hvec[script_name].iov_base = wsgi_req->script_name;
						wsgi_req->hvec[script_name].iov_len = wsgi_req->script_name_len;

						wsgi_req->hvec[wsgi_req->path_info_pos].iov_base = wsgi_req->path_info;
						wsgi_req->hvec[wsgi_req->path_info_pos].iov_len = wsgi_req->path_info_len;
#ifdef UWSGI_DEBUG
						uwsgi_log("managed SCRIPT_NAME = %.*s PATH_INFO = %.*s\n", wsgi_req->script_name_len, wsgi_req->script_name, wsgi_req->path_info_len, wsgi_req->path_info);
#endif
					} 
				}
			}
		}
	}

	
	// check for static files

	// skip extensions
	struct uwsgi_string_list *sse = uwsgi.static_skip_ext;
	while(sse) {
		if (wsgi_req->path_info_len >= sse->len) {
			if (!uwsgi_strncmp(wsgi_req->path_info+(wsgi_req->path_info_len - sse->len), sse->len, sse->value, sse->len)) {
				return 0;
			} 
		}
		sse = sse->next;
	}
	// check if a file named uwsgi.check_static+env['PATH_INFO'] exists
	udd = uwsgi.check_static;
	while(udd) {
		// need to build the path ?
		if (udd->value == NULL) {
#ifdef UWSGI_THREADING
			if (uwsgi.threads > 1) pthread_mutex_lock(&uwsgi.lock_static);
#endif
			udd->value = uwsgi_malloc(PATH_MAX+1);
			if (!realpath(udd->key, udd->value)) {
				free(udd->value);
				udd->value = NULL;
			}
#ifdef UWSGI_THREADING
			if (uwsgi.threads > 1) pthread_mutex_unlock(&uwsgi.lock_static);
#endif
			if (!udd->value) goto nextcs;				
			udd->vallen = strlen(udd->value);
		}

		if (!uwsgi_file_serve(wsgi_req, udd->value, udd->vallen, wsgi_req->path_info, wsgi_req->path_info_len, 0)) {
			return -1;
		}
nextcs:
		udd = udd->next;
	}

	// check static-map
	udd = uwsgi.static_maps;
	while(udd) {
#ifdef UWSGI_DEBUG
		uwsgi_log("checking for %.*s <-> %.*s\n", wsgi_req->path_info_len, wsgi_req->path_info, udd->keylen, udd->key);
#endif
		if (udd->status == 0) {
#ifdef UWSGI_THREADING
			if (uwsgi.threads > 1) pthread_mutex_lock(&uwsgi.lock_static);
#endif
			char *real_docroot = uwsgi_malloc(PATH_MAX+1);
			if (!realpath(udd->value, real_docroot)) {
				free(real_docroot);
				udd->value = NULL;
			}
#ifdef UWSGI_THREADING
			if (uwsgi.threads > 1) pthread_mutex_unlock(&uwsgi.lock_static);
#endif
			if (!real_docroot) goto nextsm;
			udd->value = real_docroot;
			udd->vallen = strlen(udd->value);
			udd->status = 1 + uwsgi_is_file(real_docroot);
		}

		if (!uwsgi_starts_with(wsgi_req->path_info, wsgi_req->path_info_len, udd->key, udd->keylen)) {
			if (!uwsgi_file_serve(wsgi_req, udd->value, udd->vallen, wsgi_req->path_info+udd->keylen, wsgi_req->path_info_len-udd->keylen, udd->status-1)) {
				return -1;
			}
		}
nextsm:
		udd = udd->next;
	}

	// check for static_maps in append mode
	udd = uwsgi.static_maps2;
        while(udd) {
#ifdef UWSGI_DEBUG
                uwsgi_log("checking for %.*s <-> %.*s\n", wsgi_req->path_info_len, wsgi_req->path_info, udd->keylen, udd->key);
#endif
                if (udd->status == 0) {
#ifdef UWSGI_THREADING
                        if (uwsgi.threads > 1) pthread_mutex_lock(&uwsgi.lock_static);
#endif
                        char *real_docroot = uwsgi_malloc(PATH_MAX+1);
                        if (!realpath(udd->value, real_docroot)) {
                                free(real_docroot);
                                udd->value = NULL;
                        }
#ifdef UWSGI_THREADING
                        if (uwsgi.threads > 1) pthread_mutex_unlock(&uwsgi.lock_static);
#endif
                        if (!real_docroot) goto nextsm2;
                        udd->value = real_docroot;
                        udd->vallen = strlen(udd->value);
                        udd->status = 1 + uwsgi_is_file(real_docroot);
                }

                if (!uwsgi_starts_with(wsgi_req->path_info, wsgi_req->path_info_len, udd->key, udd->keylen)) {
                        if (!uwsgi_file_serve(wsgi_req, udd->value, udd->vallen, wsgi_req->path_info, wsgi_req->path_info_len, udd->status-1)) {
                                return -1;
                        }
                }
nextsm2:
                udd = udd->next;
        }


	// finally check for docroot
	if (uwsgi.check_static_docroot && wsgi_req->document_root_len > 0) {
		char *real_docroot = uwsgi_expand_path(wsgi_req->document_root, wsgi_req->document_root_len, NULL);
		if (!real_docroot) {
                        return -1;
		}
		if (!uwsgi_file_serve(wsgi_req, real_docroot, strlen(real_docroot), wsgi_req->path_info, wsgi_req->path_info_len, 0)) {
			free(real_docroot);
                        return -1;
                }
		free(real_docroot);
	}

	return 0;
}

int uwsgi_ping_node(int node, struct wsgi_request *wsgi_req) {


	struct pollfd uwsgi_poll;

	struct uwsgi_cluster_node *ucn = &uwsgi.shared->nodes[node];

	if (ucn->name[0] == 0) {
		return 0;
	}

	if (ucn->status == UWSGI_NODE_OK) {
		return 0;
	}

	uwsgi_poll.fd = socket(AF_INET, SOCK_STREAM, 0);
	if (uwsgi_poll.fd < 0) {
		uwsgi_error("socket()");
		return -1;
	}

	if (timed_connect(&uwsgi_poll, (const struct sockaddr *) &ucn->ucn_addr, sizeof(struct sockaddr_in), uwsgi.shared->options[UWSGI_OPTION_SOCKET_TIMEOUT], 0)) {
		close(uwsgi_poll.fd);
		return -1;
	}

	wsgi_req->uh.modifier1 = UWSGI_MODIFIER_PING;
	wsgi_req->uh.pktsize = 0;
	wsgi_req->uh.modifier2 = 0;
	if (write(uwsgi_poll.fd, wsgi_req, 4) != 4) {
		uwsgi_error("write()");
		return -1;
	}

	uwsgi_poll.events = POLLIN;
	if (!uwsgi_parse_packet(wsgi_req, uwsgi.shared->options[UWSGI_OPTION_SOCKET_TIMEOUT])) {
		return -1;
	}

	return 0;
}

ssize_t uwsgi_send_empty_pkt(int fd, char *socket_name, uint8_t modifier1, uint8_t modifier2) {

	struct uwsgi_header uh;
	char *port;
	uint16_t s_port;
	struct sockaddr_in uaddr;
	int ret;

	uh.modifier1 = modifier1;
	uh.pktsize = 0;
	uh.modifier2 = modifier2;

	if (socket_name) {
		port = strchr(socket_name, ':');
		if (!port) return -1;
		s_port = atoi(port+1);
		port[0] = 0;
		memset(&uaddr, 0, sizeof(struct sockaddr_in));
		uaddr.sin_family = AF_INET;
		uaddr.sin_addr.s_addr = inet_addr(socket_name);
		uaddr.sin_port = htons(s_port);

		port[0] = ':';

		ret = sendto(fd, &uh, 4, 0, (struct sockaddr *) &uaddr, sizeof(struct sockaddr_in));
	}
	else {
		ret = send(fd, &uh, 4, 0);
	}

	if (ret < 0) {
		uwsgi_error("sendto()");
	}

	return ret;
}

int uwsgi_get_dgram(int fd, struct wsgi_request *wsgi_req) {

	ssize_t rlen;
	struct uwsgi_header *uh;
	static char *buffer = NULL;

	struct sockaddr_in sin;
	socklen_t sin_len = sizeof(struct sockaddr_in);

	if (!buffer) {
		buffer = uwsgi_malloc(uwsgi.buffer_size + 4);
	}
		

	rlen = recvfrom(fd, buffer, uwsgi.buffer_size + 4, 0, (struct sockaddr *) &sin, &sin_len);

        if (rlen < 0) {
                uwsgi_error("recvfrom");
                return -1;
        }

#ifdef UWSGI_DEBUG
	uwsgi_log("received request from %s\n", inet_ntoa(sin.sin_addr));
#endif

        if (rlen < 4) {
                uwsgi_log("invalid uwsgi packet\n");
                return -1;
        }

	uh = (struct uwsgi_header *) buffer;

	wsgi_req->uh.modifier1 = uh->modifier1;
	/* big endian ? */
#ifdef __BIG_ENDIAN__
	uh->pktsize = uwsgi_swap16(uh->pktsize);
#endif
	wsgi_req->uh.pktsize = uh->pktsize;
	wsgi_req->uh.modifier2 = uh->modifier2;

	if (wsgi_req->uh.pktsize > uwsgi.buffer_size) {
		uwsgi_log("invalid uwsgi packet size, probably you need to increase buffer size\n");
		return -1;
	}

	wsgi_req->buffer = buffer+4;

#ifdef UWSGI_DEBUG
	uwsgi_log("request received %d %d\n", wsgi_req->uh.modifier1, wsgi_req->uh.modifier2);
#endif

	return 0;

}

int uwsgi_hooked_parse(char *buffer, size_t len, void (*hook)(char *, uint16_t, char *, uint16_t, void*), void *data) {

	char *ptrbuf, *bufferend;
        uint16_t keysize = 0, valsize = 0;
        char *key;

	ptrbuf = buffer;
	bufferend = buffer + len;

	while (ptrbuf < bufferend) {
                if (ptrbuf + 2 >= bufferend) return -1;
                memcpy(&keysize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
                keysize = uwsgi_swap16(keysize);
#endif
                /* key cannot be null */
                if (!keysize)  return -1;

                ptrbuf += 2;
                if (ptrbuf + keysize > bufferend) return -1;

                // key
                key = ptrbuf;
                ptrbuf += keysize;
                // value can be null
                if (ptrbuf + 2 > bufferend) return -1;

                memcpy(&valsize, ptrbuf, 2);
#ifdef __BIG_ENDIAN__
                valsize = uwsgi_swap16(valsize);
#endif
                ptrbuf += 2;
                if (ptrbuf + valsize > bufferend) return -1;

                // now call the hook
                hook(key, keysize, ptrbuf, valsize, data);
                ptrbuf += valsize;
        }

        return 0;

}

int uwsgi_hooked_parse_dict_dgram(int fd, char *buffer, size_t len, uint8_t modifier1, uint8_t modifier2, void (*hook)(char *, uint16_t, char *, uint16_t, void *), void *data) {

	struct uwsgi_header *uh;
	ssize_t rlen;

	struct sockaddr_in sin;
	socklen_t sin_len = sizeof(struct sockaddr_in);

	char *ptrbuf, *bufferend;

	rlen = recvfrom(fd, buffer, len, 0, (struct sockaddr *) &sin, &sin_len);

	if (rlen < 0) {
		uwsgi_error("recvfrom()");
		return -1;
	}


#ifdef UWSGI_DEBUG
	uwsgi_log("RLEN: %d\n", rlen);
#endif

	// check for valid dict 4(header) 2(non-zero key)+1 2(value)
	if (rlen < (4+2+1+2)) {
#ifdef UWSGI_DEBUG
		uwsgi_log("invalid uwsgi dictionary\n");
#endif
		return -1;
	}

	uwsgi_log("received message from %s\n", inet_ntoa(sin.sin_addr));
	
	uh = (struct uwsgi_header *) buffer;

	if (uh->modifier1 != modifier1 || uh->modifier2 != modifier2) {
#ifdef UWSGI_DEBUG
		uwsgi_log("invalid uwsgi dictionary received, modifier1: %d modifier2: %d\n", uh->modifier1, uh->modifier2);
#endif
		return -1;
	}

        ptrbuf = buffer + 4;

	/* big endian ? */
#ifdef __BIG_ENDIAN__
	uh->pktsize = uwsgi_swap16(uh->pktsize);
#endif

	if (uh->pktsize > len) {
		uwsgi_log("* WARNING * the uwsgi dictionary received is too big, data will be truncated\n");
		bufferend = ptrbuf + len;
	}
	else {
        	bufferend = ptrbuf + uh->pktsize;
	}

	
#ifdef UWSGI_DEBUG
	uwsgi_log("%p %p %d\n", ptrbuf, bufferend, bufferend-ptrbuf);
#endif

	uwsgi_hooked_parse(ptrbuf, bufferend-ptrbuf, hook, data);

	return 0;

}

int uwsgi_string_sendto(int fd, uint8_t modifier1, uint8_t modifier2, struct sockaddr *sa, socklen_t sa_len, char *message, size_t len) {

	ssize_t rlen ;
	struct uwsgi_header *uh;
	char *upkt = uwsgi_malloc(len + 4);

	uh = (struct uwsgi_header *) upkt;

	uh->modifier1 = modifier1;
	uh->pktsize = len;
#ifdef __BIG_ENDIAN__
	uh->pktsize = uwsgi_swap16(uh->pktsize);
#endif
	uh->modifier2 = modifier2;

	memcpy(upkt+4, message, len);

	rlen = sendto(fd, upkt, len+4, 0, sa, sa_len);

	if (rlen < 0) {
		uwsgi_error("sendto()");
	}

	free(upkt);

	return rlen;
}

ssize_t fcgi_send_param(int fd, char *key, uint16_t keylen, char *val, uint16_t vallen) {

	struct fcgi_record fr;
	struct iovec iv[5];

	uint8_t ks1 = 0;
	uint32_t ks4 = 0;

	uint8_t vs1 = 0;
	uint32_t vs4 = 0;

	uint16_t size = keylen+vallen;

	if (keylen > 127) {
		size += 4;
		ks4 = htonl(keylen) | 0x80000000;
		iv[1].iov_base = &ks4;
		iv[1].iov_len = 4;
	}
	else {
		size += 1;
		ks1 = keylen;
		iv[1].iov_base = &ks1;
		iv[1].iov_len = 1;
	}

	if (vallen > 127) {
		size += 4;
		vs4 = htonl(vallen) | 0x80000000;
		iv[2].iov_base = &vs4;
		iv[2].iov_len = 4;
	}
	else {
		size += 1;
		vs1 = vallen;
		iv[2].iov_base = &vs1;
		iv[2].iov_len = 1;
	}

	iv[3].iov_base = key;
	iv[3].iov_len = keylen;
	iv[4].iov_base = val;
	iv[4].iov_len = vallen;

	fr.version = 1;
	fr.type = 4;
	fr.req1 = 0;
	fr.req0 = 1;
	fr.cl8.cl1 = (uint8_t) ((size >> 8) & 0xff);
	fr.cl8.cl0 = (uint8_t) (size &0xff);
	fr.pad = 0;
	fr.reserved = 0;

	iv[0].iov_base = &fr;
	iv[0].iov_len = 8;

	return writev(fd, iv, 5);
	
}

ssize_t fcgi_send_record(int fd, uint8_t type, uint16_t size, char *buffer) {

	struct fcgi_record fr;
	struct iovec iv[2];

	fr.version = 1;
	fr.type = type;
	fr.req1 = 0;
	fr.req0 = 1;
	fr.cl8.cl1 = (uint8_t) ((size >> 8) & 0xff);
	fr.cl8.cl0 = (uint8_t) (size &0xff);
	fr.pad = 0;
	fr.reserved = 0;

	iv[0].iov_base = &fr;
	iv[0].iov_len = 8;

	iv[1].iov_base = buffer;
	iv[1].iov_len = size;

	return writev(fd, iv, 2);

}

uint16_t fcgi_get_record(int fd, char *buf) {

	struct fcgi_record fr;
	uint16_t remains = 8;
	char *ptr = (char *) &fr;
	ssize_t len;

        while(remains) {
        	uwsgi_waitfd(fd, -1);
                len = read(fd, ptr, remains);
                if (len <= 0) return 0;
                remains -= len;
                ptr += len;
        }

        remains = ntohs(fr.cl) + fr.pad;
        ptr = buf;

        while(remains) {
        	uwsgi_waitfd(fd, -1);
                len = read(fd, ptr, remains);
                if (len <= 0) return 0;
                remains -= len;
                ptr += len;
        }

	if (fr.type != 6) return 0;

	return ntohs(fr.cl);

}

char *uwsgi_simple_message_string(char *socket_name, uint8_t modifier1, uint8_t modifier2, char *what, uint16_t what_len, char *buffer, uint16_t *response_len, int timeout) {

	struct wsgi_request msg_req;

	int fd = uwsgi_connect(socket_name, timeout, 0);

	if (fd < 0) {
		if (response_len) *response_len = 0;
		return NULL;
	}

	if (uwsgi_send_message(fd, modifier1, modifier2, what, what_len, -1, 0, timeout) <= 0) {
		close(fd);
		if (response_len) *response_len = 0;
		return NULL;
	}

	memset(&msg_req, 0, sizeof(struct wsgi_request));
	msg_req.poll.fd = fd;
	msg_req.poll.events = POLLIN;
	msg_req.buffer = buffer;

	if (buffer) {
		if (!uwsgi_parse_packet(&msg_req, timeout)) {
			close(fd);
			if (response_len) *response_len = 0;
			return NULL;
		}

		if (response_len) *response_len = msg_req.uh.pktsize;
	}

	close(fd);
	return buffer;
}

int uwsgi_simple_send_string2(char *socket_name, uint8_t modifier1, uint8_t modifier2, char *item1, uint16_t item1_len, char *item2, uint16_t item2_len, int timeout) {

	struct uwsgi_header uh;
	char strsize1[2], strsize2[2];

	struct iovec iov[5];

	int fd = uwsgi_connect(socket_name, timeout, 0);

        if (fd < 0) {
                return -1;
        }

	uh.modifier1 = modifier1;
	uh.pktsize = 2+item1_len+2+item2_len;
	uh.modifier2 = modifier2;

        strsize1[0] = (uint8_t) (item1_len & 0xff);
	strsize1[1] = (uint8_t) ((item1_len >> 8) & 0xff);

        strsize2[0] = (uint8_t) (item2_len & 0xff);
	strsize2[1] = (uint8_t) ((item2_len >> 8) & 0xff);

	iov[0].iov_base = &uh;
	iov[0].iov_len = 4;

	iov[1].iov_base = strsize1;
	iov[1].iov_len = 2;

	iov[2].iov_base = item1;
	iov[2].iov_len = item1_len;

	iov[3].iov_base = strsize2;
	iov[3].iov_len = 2;

	iov[4].iov_base = item2;
	iov[4].iov_len = item2_len;

	if (writev(fd, iov, 5) < 0) {
		uwsgi_error("writev()");
	}

	close(fd);
	
	return 0;
}

char *uwsgi_req_append(struct wsgi_request *wsgi_req, char *key, uint16_t keylen, char *val, uint16_t vallen) {

	if (wsgi_req->uh.pktsize + (2+keylen+2+vallen) > uwsgi.buffer_size) {
		uwsgi_log("not enough buffer space to add %.*s variable, consider increasing it with the --buffer-size option\n", keylen, key);
		return NULL;
	}

	char *ptr = wsgi_req->buffer + wsgi_req->uh.pktsize;

	*ptr++= (uint8_t) (keylen & 0xff);
        *ptr++= (uint8_t) ((keylen >> 8) & 0xff);

	memcpy(ptr, key, keylen); ptr+=keylen;

        *ptr++= (uint8_t) (vallen & 0xff);
        *ptr++= (uint8_t) ((vallen >> 8) & 0xff);

	memcpy(ptr, val, vallen);

	wsgi_req->uh.pktsize += (2+keylen+2+vallen);

	return ptr;
}

int uwsgi_simple_send_string(char *socket_name, uint8_t modifier1, uint8_t modifier2, char *item1, uint16_t item1_len, int timeout) {

        struct uwsgi_header uh;
        char strsize1[2];

        struct iovec iov[3];

        int fd = uwsgi_connect(socket_name, timeout, 0);

        if (fd < 0) {
                return -1;
        }

        uh.modifier1 = modifier1;
        uh.pktsize = 2+item1_len;
        uh.modifier2 = modifier2;

        strsize1[0] = (uint8_t) (item1_len & 0xff);
        strsize1[1] = (uint8_t) ((item1_len >> 8) & 0xff);

        iov[0].iov_base = &uh;
        iov[0].iov_len = 4;

        iov[1].iov_base = strsize1;
        iov[1].iov_len = 2;

        iov[2].iov_base = item1;
        iov[2].iov_len = item1_len;

        if (writev(fd, iov, 3) < 0) {
                uwsgi_error("writev()");
        }

        close(fd);

        return 0;
}

char *uwsgi_get_mime_type(char *name, int namelen, int *size) {

	int i;
	int count = 0;
	char *ext = NULL;
	for(i=namelen-1;i>=0;i--) {
		if (!isalnum( (int)name[i])) {
			if (name[i] == '.') {
				ext = name+(namelen-count);
				break;
			}
		}
		count++;
	}

	if (!ext) return NULL;

	struct uwsgi_dyn_dict *udd = uwsgi.mimetypes;
	while(udd) {
		if (!uwsgi_strncmp(ext, count, udd->key, udd->keylen)) {
			udd->hits++;
			// auto optimization
			if (udd->prev) {
				if (udd->hits > udd->prev->hits) {
					struct uwsgi_dyn_dict *udd_parent = udd->prev->prev, *udd_prev = udd->prev;
					if (udd_parent) {
						udd_parent->next = udd;
					}

					if (udd->next) {
						udd->next->prev = udd_prev;
					}

					udd_prev->prev = udd;
					udd_prev->next = udd->next;

					udd->prev = udd_parent;
					udd->next = udd_prev;

					if (udd->prev == NULL) {
						uwsgi.mimetypes = udd;
					}
				}
			}
			*size = udd->vallen;
			return udd->value;
		}
		udd = udd->next;
	}

	return NULL;
}

int uwsgi_append_static_path(char *dir, char *file, int file_len) {

	size_t len = strlen(dir);
	
	if (len + 1 + file_len > PATH_MAX) {
		return -1;
	}

	if (dir[len-1] == '/') {
		memcpy(dir+len, file, file_len);
		dir[len+file_len] = 0;
	}
	else {
		dir[len] = '/';
		memcpy(dir+len+1, file, file_len);
		dir[len+1+file_len] = 0;
	}

	return len;
}

int uwsgi_static_stat(char *filename, struct stat *st) {

	int ret = stat(filename, st);
	// if non-existant return -1
	if (ret < 0) return -1;

	if (S_ISREG(st->st_mode)) return 0;

	// check for index
	if (S_ISDIR(st->st_mode)) {
		struct uwsgi_string_list *usl = uwsgi.static_index;
		while(usl) {
			ret = uwsgi_append_static_path(filename, usl->value, usl->len);
			if (ret >= 0) {
#ifdef UWSGI_DEBUG
				uwsgi_log("checking for %s\n", filename);
#endif
				if (!uwsgi_static_stat(filename, st)) {
					return 0;
				}
				// reset to original name
				filename[ret] = 0;
			}
			usl = usl->next;
		}
	}

	return -1;
}

int uwsgi_real_file_serve(struct wsgi_request *wsgi_req, char *real_filename, size_t real_filename_len, struct stat *st) {

	struct iovec headers_vec[8];

	int mime_type_size = 0;
        char http_last_modified[49];

	char content_length[sizeof(UMAX64_STR)+1];

#ifdef UWSGI_THREADING
		if (uwsgi.threads > 1) pthread_mutex_lock(&uwsgi.lock_static);
#endif
		char *mime_type = uwsgi_get_mime_type(real_filename, real_filename_len, &mime_type_size);
#ifdef UWSGI_THREADING
		if (uwsgi.threads > 1) pthread_mutex_unlock(&uwsgi.lock_static);
#endif

                if (wsgi_req->if_modified_since_len) {
                        time_t ims = parse_http_date(wsgi_req->if_modified_since, wsgi_req->if_modified_since_len);
                        if (st->st_mtime <= ims) {
                                wsgi_req->status = 304;
				headers_vec[0].iov_base = wsgi_req->protocol;
				headers_vec[0].iov_len = wsgi_req->protocol_len;
				headers_vec[1].iov_base = " 304 Not Modified\r\n";
				headers_vec[1].iov_len = 19;

				wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 2);

				struct uwsgi_string_list *ah = uwsgi.additional_headers;
				while(ah) {
					headers_vec[0].iov_base = ah->value;
					headers_vec[0].iov_len = ah->len;
					headers_vec[1].iov_base = "\r\n";
					headers_vec[1].iov_len = 2;
					wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 2);
					wsgi_req->header_cnt++;
					ah = ah->next;
				}

				wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, "\r\n", 2);
                                return 0;
                        }
                }
#ifdef UWSGI_DEBUG
                uwsgi_log("[uwsgi-fileserve] file %s found\n", real_filename);
#endif

		// HTTP status
		headers_vec[0].iov_base = wsgi_req->protocol;
                headers_vec[0].iov_len = wsgi_req->protocol_len;
		headers_vec[1].iov_base = " 200 OK\r\n";
                headers_vec[1].iov_len = 9;
                wsgi_req->headers_size = wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 2);

			// uWSGI additional headers
			struct uwsgi_string_list *ah = uwsgi.additional_headers;
			while(ah) {
				headers_vec[0].iov_base = ah->value;
				headers_vec[0].iov_len = ah->len;
				headers_vec[1].iov_base = "\r\n";
				headers_vec[1].iov_len = 2;
				wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 2);
				wsgi_req->header_cnt++;
				ah = ah->next;
			}

#ifdef UWSGI_PCRE
			uwsgi_add_expires(wsgi_req, real_filename, real_filename_len, st);
#endif

			// Content-Type (if available)
			if (mime_type_size > 0 && mime_type) {
				headers_vec[0].iov_base = "Content-Type: ";
				headers_vec[0].iov_len = 14;
				headers_vec[1].iov_base = mime_type;
				headers_vec[1].iov_len = mime_type_size;
				headers_vec[2].iov_base = "\r\n";
				headers_vec[2].iov_len = 2;
				wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 3);
				wsgi_req->header_cnt++;

				// check for content-type related headers
				uwsgi_add_expires_type(wsgi_req, mime_type, mime_type_size, st);
			}

			// nginx
			if (uwsgi.file_serve_mode == 1) {
				headers_vec[0].iov_base = "X-Accel-Redirect: "; headers_vec[0].iov_len = 18 ;
				headers_vec[1].iov_base = real_filename; headers_vec[1].iov_len = real_filename_len;
				headers_vec[2].iov_base = "\r\n"; headers_vec[2].iov_len = 2;
				wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 3);
				// this is the final header (\r\n added)
                        	set_http_date(st->st_mtime, "Last-Modified", 13, http_last_modified, 1);
                        	wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, http_last_modified, 48);
                                wsgi_req->header_cnt += 2;
			}
			// apache
			else if (uwsgi.file_serve_mode == 2) {
				headers_vec[0].iov_base = "X-Sendfile: "; headers_vec[0].iov_len = 12 ;
				headers_vec[1].iov_base = real_filename; headers_vec[1].iov_len = real_filename_len;
				headers_vec[2].iov_base = "\r\n"; headers_vec[2].iov_len = 2;
				wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 3);
				// this is the final header (\r\n added)
                        	set_http_date(st->st_mtime, "Last-Modified", 13,  http_last_modified, 1);
                        	wsgi_req->headers_size += wsgi_req->socket->proto_write_header(wsgi_req, http_last_modified, 48);
                                wsgi_req->header_cnt += 2;
			}
			// raw
			else {
                        	// set Content-Length
				headers_vec[0].iov_base = "Content-Length: ";
				headers_vec[0].iov_len = 16;
				headers_vec[1].iov_len = uwsgi_long2str2n(st->st_size, content_length, sizeof(UMAX64_STR)+1);
				headers_vec[1].iov_base = content_length;
				headers_vec[2].iov_base = "\r\n";
				headers_vec[2].iov_len = 2;
				// this is the final header (\r\n added)
                        	set_http_date(st->st_mtime, "Last-Modified", 13,  http_last_modified, 1);
				headers_vec[3].iov_base = http_last_modified;
				headers_vec[3].iov_len = 48;
                        	wsgi_req->headers_size += wsgi_req->socket->proto_writev_header(wsgi_req, headers_vec, 4);
                                wsgi_req->header_cnt += 2;
                                wsgi_req->sendfile_fd = open(real_filename, O_RDONLY);
                                wsgi_req->response_size += uwsgi_sendfile(wsgi_req);
				// here we need to close the sendfile fd (no-GC involved)
				close(wsgi_req->sendfile_fd);
			}


                        wsgi_req->status = 200;
                        return 0;
}

void *uwsgi_static_offload_thread(void *req) {

	struct uwsgi_offload_request *uof_req = (struct uwsgi_offload_request *) req;

	uwsgi_real_file_serve(&uof_req->wsgi_req, uof_req->real_filename, uof_req->real_filename_len, &uof_req->st);

	// close the connection with the webserver
        if (!uof_req->wsgi_req.fd_closed || !uof_req->wsgi_req.body_as_file) {
                // NOTE, if we close the socket before receiving eventually sent data, socket layer will send a RST
                uof_req->wsgi_req.socket->proto_close(&uof_req->wsgi_req);
        }

	// free buffer
	free(uof_req->buffer);
	// free hvec
	free(uof_req->hvec);
	free(uof_req);

	pthread_mutex_lock(&uwsgi.static_offload_thread_lock);
	uwsgi.workers[uwsgi.mywid].static_offload_threads--;
	pthread_mutex_unlock(&uwsgi.static_offload_thread_lock);

	return NULL;
}
 

int uwsgi_file_serve(struct wsgi_request *wsgi_req, char *document_root, uint16_t document_root_len, char *path_info, uint16_t path_info_len, int is_a_file) {

        struct stat st;
        char real_filename[PATH_MAX+1];
	size_t real_filename_len = 0;
	char *filename = NULL;


	if (!is_a_file) {
        	filename = uwsgi_concat3n(document_root, document_root_len, "/", 1, path_info, path_info_len);
	}
	else {
		filename = uwsgi_concat2n(document_root, document_root_len, "", 0);
	}
	
#ifdef UWSGI_DEBUG
        uwsgi_log("[uwsgi-fileserve] checking for %s\n", filename);
#endif
        if (!realpath(filename, real_filename)) {
#ifdef UWSGI_DEBUG
                uwsgi_log("[uwsgi-fileserve] unable to get realpath() of the static file\n");
#endif
                free(filename);
                return -1;
        }

        free(filename);

        if (uwsgi_starts_with(real_filename, strlen(real_filename), document_root, document_root_len)) {
                uwsgi_log("[uwsgi-fileserve] security error: %s is not under %.*s\n", real_filename, document_root_len, document_root);
                return -1;
        }

        if (!uwsgi_static_stat(real_filename, &st)) {

		real_filename_len = strlen(real_filename);

		// check for skippable ext
		struct uwsgi_string_list *sse = uwsgi.static_skip_ext;
        	while(sse) {
                	if (real_filename_len >= sse->len) {
                        	if (!uwsgi_strncmp(real_filename+(real_filename_len - sse->len), sse->len, sse->value, sse->len)) {
#ifdef UWSGI_ROUTING
					if (uwsgi_apply_routes_fast(wsgi_req, real_filename, real_filename_len) == UWSGI_ROUTE_BREAK) return 0;
#endif
					return -1;
                        	}
                	}
                	sse = sse->next;
        	}

		// Ok, the file must be served as static from uWSGI
		if (uwsgi.static_offload_to_thread) {
			pthread_mutex_lock(&uwsgi.static_offload_thread_lock);
			uint64_t offload_thread_count = uwsgi.workers[uwsgi.mywid].static_offload_threads;
			pthread_mutex_unlock(&uwsgi.static_offload_thread_lock);

			if (offload_thread_count > (uint64_t) uwsgi.static_offload_to_thread) {
				uwsgi_log_verbose("OVERLOAD !!! unable to offload static file serving\n");
				return uwsgi_real_file_serve(wsgi_req, real_filename, real_filename_len, &st);
			}
			
			struct uwsgi_offload_request *uor = uwsgi_malloc(sizeof(struct uwsgi_offload_request));

			// buffer
			uor->buffer = uwsgi_malloc(uwsgi.buffer_size); memcpy(uor->buffer, wsgi_req->buffer, uwsgi.buffer_size);
			// iovec
			uor->hvec = uwsgi_malloc(sizeof(struct iovec) * uwsgi.vec_size); memcpy(uor->hvec, wsgi_req->hvec, sizeof(struct iovec) * uwsgi.vec_size);

			// wsgi_req
			memcpy(&uor->wsgi_req, wsgi_req, sizeof(struct wsgi_request));

			uor->wsgi_req.buffer = uor->buffer;
			uor->wsgi_req.hvec = uor->hvec;

			// stat
			memcpy(&uor->st, &st, sizeof(struct stat));

			// filename
			memcpy(uor->real_filename, real_filename, real_filename_len);
			uor->real_filename_len = real_filename_len;
			uor->real_filename[uor->real_filename_len] = 0;

			// avoid closing the connection
			wsgi_req->fd_closed = 1;

			pthread_mutex_lock(&uwsgi.static_offload_thread_lock);
			uwsgi.workers[uwsgi.mywid].static_offload_threads++;
			pthread_mutex_unlock(&uwsgi.static_offload_thread_lock);

			if (pthread_create(&uor->tid, &uwsgi.static_offload_thread_attr, uwsgi_static_offload_thread, (void *) uor)) {
				uwsgi_error("pthread_create()");
				// bad condition, better to exit...
				exit(1);
			}
			wsgi_req->status = -30;
			return 0;
		}
		else {
			return uwsgi_real_file_serve(wsgi_req, real_filename, real_filename_len, &st);
		}
        }

        return -1;

}
