(strip_return_port): Patch from "alex" to strip the port from the host

string and return the port.  I cleaned up and added error handling to
the code, but it's basically "alex"'s fix.

(extract_http_url): Rewrote this function to remove all the sscanf()
calls.  It's much easier to just split on the path slash (if it's
present) and then strip the user name/password and port from the host
string.  Less code, handles more cases!
This commit is contained in:
Robert James Kaes 2004-02-04 19:57:40 +00:00
parent 47df93b8f3
commit bf22966f55

View File

@ -1,4 +1,4 @@
/* $Id: reqs.c,v 1.109 2004-01-26 19:11:51 rjkaes Exp $
/* $Id: reqs.c,v 1.110 2004-02-04 19:57:40 rjkaes Exp $
*
* This is where all the work in tinyproxy is actually done. Incoming
* connections have a new child created for them. The child then
@ -217,6 +217,26 @@ strip_username_password(char* host)
}
}
/*
* Take a host string and if there is a port part, strip
* it off and set proper port variable i.e. for www.host.com:8001
*/
static int
strip_return_port(char* host)
{
char *ptr1;
int port;
ptr1 = strchr(host, ':');
if (ptr1 == NULL)
return 0;
*ptr1++ = '\0';
if (sscanf(ptr1, "%d", &port) != 1) /* one conversion required */
return 0;
return port;
}
/*
* Pull the information out of the URL line. This will handle both HTTP
* and FTP (proxied) URLs.
@ -224,31 +244,33 @@ strip_username_password(char* host)
static int
extract_http_url(const char *url, struct request_s *request)
{
request->host = (char*)safemalloc(strlen(url) + 1);
request->path = (char*)safemalloc(strlen(url) + 1);
char *p;
int len;
int port;
/* Split the URL on the slash to separate host from path */
p = strchr(url, '/');
if (p != NULL) {
len = p - url;
request->host = safemalloc(len + 1);
memcpy(request->host, url, len);
request->host[len] = '\0';
request->path = safestrdup(p);
} else {
request->host = safestrdup(url);
request->path = safestrdup("/");
}
if (!request->host || !request->path)
goto ERROR_EXIT;
if (sscanf
(url, "%[^:/]:%hu%s", request->host, &request->port,
request->path) == 3) ;
else if (sscanf(url, "%[^/]%s", request->host, request->path) == 2)
request->port = HTTP_PORT;
else if (sscanf(url, "%[^:/]:%hu", request->host, &request->port)
== 2)
strcpy(request->path, "/");
else if (sscanf(url, "%[^/]", request->host) == 1) {
request->port = HTTP_PORT;
strcpy(request->path, "/");
} else {
log_message(LOG_ERR, "extract_http_url: Can't parse URL.");
goto ERROR_EXIT;
}
/* Remove the username/password if they're present */
strip_username_password(request->host);
/* Find a proper port in www.site.com:8001 URLs */
port = strip_return_port(request->host);
request->port = (port != 0) ? port : HTTP_PORT;
return 0;
ERROR_EXIT: