jscoverage/trunk/http-connection.c

/*
    http-connection.c - TCP connection between HTTP client and server
    Copyright (C) 2008, 2009 siliconforks.com

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License along
    with this program; if not, write to the Free Software Foundation, Inc.,
    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

#include <config.h>

#include "http-server.h"

#include <assert.h>
#include <string.h>

#include "util.h"

#define CONNECTION_BUFFER_CAPACITY 8192

#ifdef _WIN32
#define ERRNO (WSAGetLastError())
#else
#include <errno.h>
#define ERRNO errno
#endif

struct HTTPConnection {
  SOCKET s;
  uint8_t input_buffer[CONNECTION_BUFFER_CAPACITY];
  size_t input_buffer_offset;
  size_t input_buffer_length;
  uint8_t output_buffer[CONNECTION_BUFFER_CAPACITY];
  size_t output_buffer_offset;
  size_t output_buffer_length;
};

static HTTPConnection * HTTPConnection_new(SOCKET s) {
  HTTPConnection * connection = xmalloc(sizeof(HTTPConnection));
  connection->s = s;
  connection->input_buffer_offset = 0;
  connection->input_buffer_length = 0;
  connection->output_buffer_offset = 0;
  connection->output_buffer_length = 0;
  return connection;
}

HTTPConnection * HTTPConnection_new_client(const char * host, uint16_t port) {
  struct in_addr ip_address;
  if (! inet_aton(host, &ip_address)) {
    /* it's a host name */
    if (xgethostbyname(host, &ip_address) != 0) {
      return NULL;
    }
  }

  SOCKET s = socket(PF_INET, SOCK_STREAM, 0);
  if (s == INVALID_SOCKET) {
    return NULL;
  }

  struct sockaddr_in a;
  a.sin_family = AF_INET;
  a.sin_port = htons(port);
  a.sin_addr = ip_address;

  if (connect(s, (struct sockaddr *) &a, sizeof(a)) < 0) {
    closesocket(s);
    return NULL;
  }

  return HTTPConnection_new(s);
}

HTTPConnection * HTTPConnection_new_server(SOCKET s) {
  return HTTPConnection_new(s);
}

int HTTPConnection_delete(HTTPConnection * connection) {
  int result = 0;
  if (closesocket(connection->s) == -1) {
    result = ERRNO;
    assert(result != 0);
  }
  free(connection);
  return result;
}

int HTTPConnection_get_peer(HTTPConnection * connection, struct sockaddr_in * peer) {
  int result = 0;
  socklen_t length = sizeof(struct sockaddr_in);
  if (getpeername(connection->s, (struct sockaddr *) peer, &length) == -1) {
    result = ERRNO;
    assert(result != 0);
  }
  return result;
}

int HTTPConnection_read_octet(HTTPConnection * connection, int * octet) {
  if (connection->input_buffer_offset >= connection->input_buffer_length) {
    ssize_t bytes_received = recv(connection->s, connection->input_buffer, CONNECTION_BUFFER_CAPACITY, 0);
    if (bytes_received == -1) {
      int result = ERRNO;
      assert(result != 0);
      return result;
    }
    else if (bytes_received == 0) {
      /* orderly shutdown */
      *octet = -1;
      return 0;
    }
    else {
      connection->input_buffer_offset = 0;
      connection->input_buffer_length = bytes_received;
    }
  }
  *octet = connection->input_buffer[connection->input_buffer_offset];
  connection->input_buffer_offset++;
  return 0;
}

int HTTPConnection_peek_octet(HTTPConnection * connection, int * octet) {
  int result = HTTPConnection_read_octet(connection, octet);

  /* check for error */
  if (result != 0) {
    return result;
  }

  /* check for end */
  if (*octet == -1) {
    return 0;
  }

  /* reset input buffer */
  connection->input_buffer_offset--;
  return 0;
}

int HTTPConnection_write(HTTPConnection * connection, const void * p, size_t size) {
  while (size > 0) {
    if (connection->output_buffer_length == CONNECTION_BUFFER_CAPACITY) {
      /* buffer full */
      ssize_t bytes_sent = send(connection->s, connection->output_buffer, CONNECTION_BUFFER_CAPACITY, 0);
      if (bytes_sent == -1) {
        int result = ERRNO;
        assert(result != 0);
        return result;
      }
      connection->output_buffer_length = 0;
    }

    size_t buffer_remaining = CONNECTION_BUFFER_CAPACITY - connection->output_buffer_length;
    size_t bytes_to_copy;
    if (size <= buffer_remaining) {
      bytes_to_copy = size;
    }
    else {
      bytes_to_copy = buffer_remaining;
    }

    memcpy(connection->output_buffer + connection->output_buffer_length, p, bytes_to_copy);
    connection->output_buffer_length += bytes_to_copy;
    p += bytes_to_copy;
    size -= bytes_to_copy;
  }

  return 0;
}

int HTTPConnection_flush(HTTPConnection * connection) {
  if (connection->output_buffer_length > 0) {
    ssize_t bytes_sent = send(connection->s, connection->output_buffer, connection->output_buffer_length, 0);
    if (bytes_sent == -1) {
      int result = ERRNO;
      assert(result != 0);
      return result;
    }
    connection->output_buffer_length = 0;
  }
  return 0;
}
1	/*
2	http-connection.c - TCP connection between HTTP client and server
3	Copyright (C) 2008, 2009 siliconforks.com
4
5	This program is free software; you can redistribute it and/or modify
6	it under the terms of the GNU General Public License as published by
7	the Free Software Foundation; either version 2 of the License, or
8	(at your option) any later version.
9
10	This program is distributed in the hope that it will be useful,
11	but WITHOUT ANY WARRANTY; without even the implied warranty of
12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13	GNU General Public License for more details.
14
15	You should have received a copy of the GNU General Public License along
16	with this program; if not, write to the Free Software Foundation, Inc.,
17	51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18	*/
19
20	#include <config.h>
21
22	#include "http-server.h"
23
24	#include <assert.h>
25	#include <string.h>
26
27	#include "util.h"
28
29	#define CONNECTION_BUFFER_CAPACITY 8192
30
31	#ifdef _WIN32
32	#define ERRNO (WSAGetLastError())
33	#else
34	#include <errno.h>
35	#define ERRNO errno
36	#endif
37
38	struct HTTPConnection {
39	SOCKET s;
40	uint8_t input_buffer[CONNECTION_BUFFER_CAPACITY];
41	size_t input_buffer_offset;
42	size_t input_buffer_length;
43	uint8_t output_buffer[CONNECTION_BUFFER_CAPACITY];
44	size_t output_buffer_offset;
45	size_t output_buffer_length;
46	};
47
48	static HTTPConnection * HTTPConnection_new(SOCKET s) {
49	HTTPConnection * connection = xmalloc(sizeof(HTTPConnection));
50	connection->s = s;
51	connection->input_buffer_offset = 0;
52	connection->input_buffer_length = 0;
53	connection->output_buffer_offset = 0;
54	connection->output_buffer_length = 0;
55	return connection;
56	}
57
58	HTTPConnection * HTTPConnection_new_client(const char * host, uint16_t port) {
59	struct in_addr ip_address;
60	if (! inet_aton(host, &ip_address)) {
61	/* it's a host name */
62	if (xgethostbyname(host, &ip_address) != 0) {
63	return NULL;
64	}
65	}
66
67	SOCKET s = socket(PF_INET, SOCK_STREAM, 0);
68	if (s == INVALID_SOCKET) {
69	return NULL;
70	}
71
72	struct sockaddr_in a;
73	a.sin_family = AF_INET;
74	a.sin_port = htons(port);
75	a.sin_addr = ip_address;
76
77	if (connect(s, (struct sockaddr *) &a, sizeof(a)) < 0) {
78	closesocket(s);
79	return NULL;
80	}
81
82	return HTTPConnection_new(s);
83	}
84
85	HTTPConnection * HTTPConnection_new_server(SOCKET s) {
86	return HTTPConnection_new(s);
87	}
88
89	int HTTPConnection_delete(HTTPConnection * connection) {
90	int result = 0;
91	if (closesocket(connection->s) == -1) {
92	result = ERRNO;
93	assert(result != 0);
94	}
95	free(connection);
96	return result;
97	}
98
99	int HTTPConnection_get_peer(HTTPConnection * connection, struct sockaddr_in * peer) {
100	int result = 0;
101	socklen_t length = sizeof(struct sockaddr_in);
102	if (getpeername(connection->s, (struct sockaddr *) peer, &length) == -1) {
103	result = ERRNO;
104	assert(result != 0);
105	}
106	return result;
107	}
108
109	int HTTPConnection_read_octet(HTTPConnection * connection, int * octet) {
110	if (connection->input_buffer_offset >= connection->input_buffer_length) {
111	ssize_t bytes_received = recv(connection->s, connection->input_buffer, CONNECTION_BUFFER_CAPACITY, 0);
112	if (bytes_received == -1) {
113	int result = ERRNO;
114	assert(result != 0);
115	return result;
116	}
117	else if (bytes_received == 0) {
118	/* orderly shutdown */
119	*octet = -1;
120	return 0;
121	}
122	else {
123	connection->input_buffer_offset = 0;
124	connection->input_buffer_length = bytes_received;
125	}
126	}
127	*octet = connection->input_buffer[connection->input_buffer_offset];
128	connection->input_buffer_offset++;
129	return 0;
130	}
131
132	int HTTPConnection_peek_octet(HTTPConnection * connection, int * octet) {
133	int result = HTTPConnection_read_octet(connection, octet);
134
135	/* check for error */
136	if (result != 0) {
137	return result;
138	}
139
140	/* check for end */
141	if (*octet == -1) {
142	return 0;
143	}
144
145	/* reset input buffer */
146	connection->input_buffer_offset--;
147	return 0;
148	}
149
150	int HTTPConnection_write(HTTPConnection * connection, const void * p, size_t size) {
151	while (size > 0) {
152	if (connection->output_buffer_length == CONNECTION_BUFFER_CAPACITY) {
153	/* buffer full */
154	ssize_t bytes_sent = send(connection->s, connection->output_buffer, CONNECTION_BUFFER_CAPACITY, 0);
155	if (bytes_sent == -1) {
156	int result = ERRNO;
157	assert(result != 0);
158	return result;
159	}
160	connection->output_buffer_length = 0;
161	}
162
163	size_t buffer_remaining = CONNECTION_BUFFER_CAPACITY - connection->output_buffer_length;
164	size_t bytes_to_copy;
165	if (size <= buffer_remaining) {
166	bytes_to_copy = size;
167	}
168	else {
169	bytes_to_copy = buffer_remaining;
170	}
171
172	memcpy(connection->output_buffer + connection->output_buffer_length, p, bytes_to_copy);
173	connection->output_buffer_length += bytes_to_copy;
174	p += bytes_to_copy;
175	size -= bytes_to_copy;
176	}
177
178	return 0;
179	}
180
181	int HTTPConnection_flush(HTTPConnection * connection) {
182	if (connection->output_buffer_length > 0) {
183	ssize_t bytes_sent = send(connection->s, connection->output_buffer, connection->output_buffer_length, 0);
184	if (bytes_sent == -1) {
185	int result = ERRNO;
186	assert(result != 0);
187	return result;
188	}
189	connection->output_buffer_length = 0;
190	}
191	return 0;
192	}