mirrors/tor
1
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/core/tor.git synced 2024-11-20 10:12:15 +01:00
Code Issues Projects Releases Wiki Activity

r11919@Kushana: nickm | 2007-01-10 13:32:48 -0500

Browse Source 
Add some defensive programming to eventdns.c in an attempt to catch possible memory stomping bugs.


svn:r9322
This commit is contained in:
Nick Mathewson 2007-01-10 19:49:21 +00:00
parent 53b730556e
commit 9937695598
2 changed files with 40 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ChangeLog
View File

@ -3,6 +3,8 @@ Changes in version 0.1.2.7-alpha - 2007-??-??
- When computing clock skew from directory HTTP headers, consider what - When computing clock skew from directory HTTP headers, consider what
time it was when we finished asking for the directory, not what time it time it was when we finished asking for the directory, not what time it
is now. is now.
- Add some defensive programming to eventdns.c in an attempt to catch
possible memory-stomping bugs.
Changes in version 0.1.2.6-alpha - 2007-01-09 Changes in version 0.1.2.6-alpha - 2007-01-09

35
src/or/eventdns.c
View File

@ -141,6 +141,8 @@ typedef unsigned int uint;
#define CLASS_INET EVDNS_CLASS_INET #define CLASS_INET EVDNS_CLASS_INET
#define CLEAR(x) do { memset((x), 0, sizeof(*(x))); } while(0)
struct request { struct request {
u8 *request; // the dns packet data u8 *request; // the dns packet data
unsigned int request_len; unsigned int request_len;
@ -450,6 +452,7 @@ static void
nameserver_probe_failed(struct nameserver *const ns) { nameserver_probe_failed(struct nameserver *const ns) {
const struct timeval * timeout; const struct timeval * timeout;
(void) evtimer_del(&ns->timeout_event); (void) evtimer_del(&ns->timeout_event);
CLEAR(&ns->timeout_event);
if (ns->state == 1) { if (ns->state == 1) {
// This can happen if the nameserver acts in a way which makes us mark // This can happen if the nameserver acts in a way which makes us mark
// it as bad and then starts sending good replies. // it as bad and then starts sending good replies.
@ -526,6 +529,7 @@ nameserver_up(struct nameserver *const ns) {
log(EVDNS_LOG_WARN, "Nameserver %s is back up", log(EVDNS_LOG_WARN, "Nameserver %s is back up",
debug_ntoa(ns->address)); debug_ntoa(ns->address));
evtimer_del(&ns->timeout_event); evtimer_del(&ns->timeout_event);
CLEAR(&ns->timeout_event);
ns->state = 1; ns->state = 1;
ns->failed_times = 0; ns->failed_times = 0;
ns->timedout = 0; ns->timedout = 0;
@ -557,6 +561,7 @@ request_finished(struct request *const req, struct request **head) {
log(EVDNS_LOG_DEBUG, "Removing timeout for request %lx", log(EVDNS_LOG_DEBUG, "Removing timeout for request %lx",
(unsigned long) req); (unsigned long) req);
evtimer_del(&req->timeout_event); evtimer_del(&req->timeout_event);
CLEAR(&req->timeout_event);
search_request_finished(req); search_request_finished(req);
global_requests_inflight--; global_requests_inflight--;
@ -569,6 +574,7 @@ request_finished(struct request *const req, struct request **head) {
// so everything gets free()ed when we: // so everything gets free()ed when we:
} }
CLEAR(req);
free(req); free(req);
evdns_requests_pump_waiting_queue(); evdns_requests_pump_waiting_queue();
@ -975,6 +981,7 @@ err:
free(server_req->base.questions[i]); free(server_req->base.questions[i]);
free(server_req->base.questions); free(server_req->base.questions);
} }
CLEAR(server_req);
free(server_req); free(server_req);
} }
return -1; return -1;
@ -1136,6 +1143,7 @@ server_port_flush(struct evdns_server_port *port)
// We have no more pending requests; stop listening for 'writeable' events. // We have no more pending requests; stop listening for 'writeable' events.
(void) event_del(&port->event); (void) event_del(&port->event);
CLEAR(&port->event);
event_set(&port->event, port->socket, EV_READ | EV_PERSIST, event_set(&port->event, port->socket, EV_READ | EV_PERSIST,
server_port_ready_callback, port); server_port_ready_callback, port);
if (event_add(&port->event, NULL) < 0) { if (event_add(&port->event, NULL) < 0) {
@ -1153,6 +1161,7 @@ nameserver_write_waiting(struct nameserver *ns, char waiting) {
ns->write_waiting = waiting; ns->write_waiting = waiting;
(void) event_del(&ns->event); (void) event_del(&ns->event);
CLEAR(&ns->event);
event_set(&ns->event, ns->socket, EV_READ | (waiting ? EV_WRITE : 0) | EV_PERSIST, event_set(&ns->event, ns->socket, EV_READ | (waiting ? EV_WRITE : 0) | EV_PERSIST,
nameserver_ready_callback, ns); nameserver_ready_callback, ns);
if (event_add(&ns->event, NULL) < 0) { if (event_add(&ns->event, NULL) < 0) {
@ -1380,6 +1389,7 @@ evdns_add_server_port(int socket, int is_tcp, evdns_request_callback_fn_type cb,
struct evdns_server_port *port; struct evdns_server_port *port;
if (!(port = malloc(sizeof(struct evdns_server_port)))) if (!(port = malloc(sizeof(struct evdns_server_port))))
return NULL; return NULL;
memset(port, 0, sizeof(struct evdns_server_port));
assert(!is_tcp); // TCP sockets not yet implemented assert(!is_tcp); // TCP sockets not yet implemented
port->socket = socket; port->socket = socket;
@ -1438,8 +1448,10 @@ evdns_server_request_add_reply(struct evdns_server_request *_req, int section, c
item = malloc(sizeof(struct server_reply_item)); item = malloc(sizeof(struct server_reply_item));
if (!item) if (!item)
return -1; return -1;
CLEAR(item);
item->next = NULL; item->next = NULL;
if (!(item->name = strdup(name))) { if (!(item->name = strdup(name))) {
CLEAR(item);
free(item); free(item);
return -1; return -1;
} }
@ -1453,6 +1465,7 @@ evdns_server_request_add_reply(struct evdns_server_request *_req, int section, c
if (item->is_name) { if (item->is_name) {
if (!(item->data = strdup(data))) { if (!(item->data = strdup(data))) {
free(item->name); free(item->name);
CLEAR(item);
free(item); free(item);
return -1; return -1;
} }
@ -1460,6 +1473,7 @@ evdns_server_request_add_reply(struct evdns_server_request *_req, int section, c
} else { } else {
if (!(item->data = malloc(datalen))) { if (!(item->data = malloc(datalen))) {
free(item->name); free(item->name);
CLEAR(item);
free(item); free(item);
return -1; return -1;
} }
@ -1650,6 +1664,7 @@ evdns_server_request_respond(struct evdns_server_request *_req, int err)
port->choked = 1; port->choked = 1;
(void) event_del(&port->event); (void) event_del(&port->event);
CLEAR(&port->event);
event_set(&port->event, port->socket, (port->closing?0:EV_READ) | EV_WRITE | EV_PERSIST, server_port_ready_callback, port); event_set(&port->event, port->socket, (port->closing?0:EV_READ) | EV_WRITE | EV_PERSIST, server_port_ready_callback, port);
if (event_add(&port->event, NULL) < 0) { if (event_add(&port->event, NULL) < 0) {
@ -1689,6 +1704,7 @@ server_request_free_answers(struct server_request *req)
free(victim->name); free(victim->name);
if (victim->data) if (victim->data)
free(victim->data); free(victim->data);
/* XXXX free(victim?) -NM */
victim = next; victim = next;
} }
*list = NULL; *list = NULL;
@ -1716,8 +1732,9 @@ server_request_free(struct server_request *req)
rc = --req->port->refcnt; rc = --req->port->refcnt;
} }
if (req->response) if (req->response) {
free(req->response); free(req->response);
}
server_request_free_answers(req); server_request_free_answers(req);
@ -1728,9 +1745,11 @@ server_request_free(struct server_request *req)
if (rc == 0) { if (rc == 0) {
server_port_free(req->port); server_port_free(req->port);
CLEAR(req);
free(req); free(req);
return (1); return (1);
} }
CLEAR(req);
free(req); free(req);
return (0); return (0);
} }
@ -1747,6 +1766,8 @@ server_port_free(struct evdns_server_port *port)
port->socket = -1; port->socket = -1;
} }
(void) event_del(&port->event); (void) event_del(&port->event);
CLEAR(&port->event);
// XXXX actually free the port? -NM
} }
// exported function // exported function
@ -1778,6 +1799,7 @@ evdns_request_timeout_callback(int fd, short events, void *arg) {
} }
(void) evtimer_del(&req->timeout_event); (void) evtimer_del(&req->timeout_event);
CLEAR(&req->timeout_event);
if (req->tx_count >= global_max_retransmits) { if (req->tx_count >= global_max_retransmits) {
// this request has failed // this request has failed
reply_callback(req, 0, DNS_ERR_TIMEOUT, NULL); reply_callback(req, 0, DNS_ERR_TIMEOUT, NULL);
@ -1938,9 +1960,12 @@ evdns_clear_nameservers_and_suspend(void)
while (1) { while (1) {
struct nameserver *next = server->next; struct nameserver *next = server->next;
(void) event_del(&server->event); (void) event_del(&server->event);
CLEAR(&server->event);
(void) evtimer_del(&server->timeout_event); (void) evtimer_del(&server->timeout_event);
CLEAR(&server->timeout_event);
if (server->socket >= 0) if (server->socket >= 0)
CLOSE_SOCKET(server->socket); CLOSE_SOCKET(server->socket);
CLEAR(server);
free(server); free(server);
if (next == started_at) if (next == started_at)
break; break;
@ -1955,6 +1980,7 @@ evdns_clear_nameservers_and_suspend(void)
req->ns = NULL; req->ns = NULL;
// ???? What to do about searches? // ???? What to do about searches?
(void) evtimer_del(&req->timeout_event); (void) evtimer_del(&req->timeout_event);
CLEAR(&req->timeout_event);
req->trans_id = 0; req->trans_id = 0;
req->transmit_me = 0; req->transmit_me = 0;
@ -2054,6 +2080,7 @@ evdns_nameserver_add(unsigned long int address) {
out2: out2:
CLOSE_SOCKET(ns->socket); CLOSE_SOCKET(ns->socket);
out1: out1:
CLEAR(ns);
free(ns); free(ns);
log(EVDNS_LOG_WARN, "Unable to add nameserver %s: error %d", debug_ntoa(address), err); log(EVDNS_LOG_WARN, "Unable to add nameserver %s: error %d", debug_ntoa(address), err);
return err; return err;
@ -2130,6 +2157,7 @@ request_new(int type, const char *name, int flags,
return req; return req;
err1: err1:
CLEAR(req);
free(req); free(req);
return NULL; return NULL;
} }
@ -2258,8 +2286,10 @@ search_state_decref(struct search_state *const state) {
struct search_domain *next, *dom; struct search_domain *next, *dom;
for (dom = state->head; dom; dom = next) { for (dom = state->head; dom; dom = next) {
next = dom->next; next = dom->next;
CLEAR(dom);
free(dom); free(dom);
} }
CLEAR(state);
free(state); free(state);
} }
} }
@ -2911,6 +2941,7 @@ evdns_shutdown(int fail_requests)
if (server->socket >= 0) if (server->socket >= 0)
CLOSE_SOCKET(server->socket); CLOSE_SOCKET(server->socket);
(void) event_del(&server->event); (void) event_del(&server->event);
CLEAR(server);
free(server); free(server);
if (server_next == server_head) if (server_next == server_head)
break; break;
@ -2921,8 +2952,10 @@ evdns_shutdown(int fail_requests)
if (global_search_state) { if (global_search_state) {
for (dom = global_search_state->head; dom; dom = dom_next) { for (dom = global_search_state->head; dom; dom = dom_next) {
dom_next = dom->next; dom_next = dom->next;
CLEAR(dom);
free(dom); free(dom);
} }
CLEAR(global_search_state);
free(global_search_state); free(global_search_state);
global_search_state = NULL; global_search_state = NULL;
} }