From 772dc6bc66e401bb6e1c27b120d6bdcee70aa193 Mon Sep 17 00:00:00 2001 From: Eric Wong Date: Mon, 19 Feb 2007 07:58:03 +0000 Subject: inputStream_http: cleanup GET request code (finishHTTPInit) Avoid unnecessary memset to zero, snprintf always puts a trailing '\0'. We also have no need to subtract one from the buffer we're snprintf-ing it to. We also check the return value of snprintf to ensure it's not too long. I have a feeling we might as well avoid snprintf altogether so we don't have to worry about buffer sizing/stack overflow and just do a bunch of write(2)s, letting Nagle sort it out... Also, centralize some of the exit error handling in with goto. This makes the code a bit more consistent and maintainable as well as reducing code and binary size. git-svn-id: https://svn.musicpd.org/mpd/trunk@5395 09075e82-0dd4-0310-85a5-a0d7c8717e4f --- src/inputStream_http.c | 58 +++++++++++++++++++++++++------------------------- 1 file changed, 29 insertions(+), 29 deletions(-) diff --git a/src/inputStream_http.c b/src/inputStream_http.c index 97be2775c..6f2f8e991 100644 --- a/src/inputStream_http.c +++ b/src/inputStream_http.c @@ -446,7 +446,8 @@ static int finishHTTPInit(InputStream * inStream) int error; socklen_t error_len = sizeof(int); int ret; - char request[2049]; + int length; + char request[2048]; tv.tv_sec = 0; tv.tv_usec = 0; @@ -463,42 +464,41 @@ static int finishHTTPInit(InputStream * inStream) if (ret < 0) { DEBUG(__FILE__ ": problem select'ing: %s\n", strerror(errno)); - close(data->sock); - data->connState = HTTP_CONN_STATE_CLOSED; - return -1; + goto close_err; } getsockopt(data->sock, SOL_SOCKET, SO_ERROR, &error, &error_len); - if (error) { - close(data->sock); - data->connState = HTTP_CONN_STATE_CLOSED; - return -1; - } + if (error) + goto close_err; - memset(request, 0, 2049); /* deal with ICY metadata later, for now its fucking up stuff! */ - snprintf(request, 2048, "GET %s HTTP/1.1\r\n" "Host: %s\r\n" - /*"Connection: close\r\n" */ - "User-Agent: %s/%s\r\n" - "Range: bytes=%ld-\r\n" - "%s" /* authorization */ - "Icy-Metadata:1\r\n" - "\r\n", data->path, data->host, PACKAGE_NAME, PACKAGE_VERSION, - inStream->offset, - data->proxyAuth ? data->proxyAuth : - (data->httpAuth ? data->httpAuth : "") - ); - - ret = write(data->sock, request, strlen(request)); - if (ret != strlen(request)) { - close(data->sock); - data->connState = HTTP_CONN_STATE_CLOSED; - return -1; - } + length = snprintf(request, sizeof(request), + "GET %s HTTP/1.1\r\n" "Host: %s\r\n" + /*"Connection: close\r\n" */ + "User-Agent: %s/%s\r\n" + "Range: bytes=%ld-\r\n" + "%s" /* authorization */ + "Icy-Metadata:1\r\n" + "\r\n", + data->path, data->host, + PACKAGE_NAME, PACKAGE_VERSION, + inStream->offset, + data->proxyAuth ? data->proxyAuth : + (data->httpAuth ? data->httpAuth : "")); + + if (length >= sizeof(request)) + goto close_err; + ret = write(data->sock, request, length); + if (ret != length) + goto close_err; data->connState = HTTP_CONN_STATE_HELLO; - return 0; + +close_err: + close(data->sock); + data->connState = HTTP_CONN_STATE_CLOSED; + return -1; } static int getHTTPHello(InputStream * inStream) -- cgit v1.2.3